package com.rabbitmq.callback;

import lombok.extern.slf4j.Slf4j;
import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;
import javax.annotation.Resource;

@Slf4j
@Component
public class ConfirmCallBack implements RabbitTemplate.ConfirmCallback,RabbitTemplate.ReturnCallback {

    @Autowired
    RabbitTemplate rabbitTemplate;

    /**
     * 这里使用Bean的后置处理
     * 保证注入了RabbitTemplate后针对ConfirmCallback设置回调的实现类
     * 否则，将无法进行回调（因为是一个接口，需要实现类）
     */
    @PostConstruct
    public void setCallBackMethodImpl(){
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }

    /**
     * 交换机不管是否收到消息的一个回调方法
     * CorrelationData
     * 消息相关数据
     * ack
     * 交换机是否收到消息
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {

        String id = correlationData != null ? correlationData.getId():"";
        if(ack){
            log.info("交换机已经收到 id 为:{}的消息",id);
        }else{
            log.info("交换机还未收到 id 为:{}消息,由于原因:{}",id,cause);
        }

    }


    /**
     * 当消息无法路由的时候的回调方法
     * 对应配置文件的 publisher-returns 参数
     * 当交换机无法获取到队列信息时,不使用该方式,则会将消息丢弃
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {
        log.error("消息 {}, 被交换机 {} 退回，退回原因 :{}, 路由 key:{}",new
                String(message.getBody()),exchange,replyText,routingKey);
    }
}
